# 定义GPU数量变量
NUM_GPUS=2

#训练代码
python train_net_lmpm.py \
    --config-file configs/tlggnv2_SWIN_bs8.yaml \
    --num-gpus $NUM_GPUS --dist-url auto \
    MODEL.WEIGHTS ./weights/model_final_86143f.pkl

echo "train end"

sleep 15

echo "test start"


#推理代码
python train_net_lmpm.py \
    --config-file configs/tlggnv2_SWIN_bs8.yaml \
    --num-gpus $NUM_GPUS --dist-url auto --eval-only \
    MODEL.WEIGHTS ./outputs/tlggnv2k20/model_final.pth DATASETS.TEST '("mevis_test",)'